51f360b1e2833f78b8515bebef5fc3d7e384e19b,h2o-algos/src/main/java/hex/deeplearning/DeepLearning.java,DeepLearning,trainModel,#DeepLearningModel#,961

Before Change


      float[] trainSamplingFactors;
      if (mp.classification && mp.balance_classes) {
        trainSamplingFactors = new float[train.lastVec().domain().length]; //leave initialized to 0 -> will be filled up below
        train = updateFrame(train, sampleFrameStratified(
                train, train.lastVec(), trainSamplingFactors, (long)(mp.max_after_balance_size*train.numRows()), mp.seed, true, false));
        model.setModelClassDistribution(new MRUtils.ClassDist(train.lastVec()).doAll(train.lastVec()).rel_dist());
      }
      model.training_rows = train.numRows();

After Change


      float[] trainSamplingFactors;
      if (mp.classification && mp.balance_classes) {
        trainSamplingFactors = new float[train.lastVec().domain().length]; //leave initialized to 0 -> will be filled up below
        train = sampleFrameStratified(
                train, train.lastVec(), trainSamplingFactors, (long)(mp.max_after_balance_size*train.numRows()), mp.seed, true, false);
        model.setModelClassDistribution(new MRUtils.ClassDist(train.lastVec()).doAll(train.lastVec()).rel_dist());
      }
      model.training_rows = train.numRows();